home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2001 December / pcwk12201b.iso / Wersje pelne i specjalne / Winamp 2.77 i 3.0beta / wasabi-sdk_beta1.exe / jnetlib / udpconnection.cpp < prev    next >
Encoding:
C/C++ Source or Header  |  2001-10-08  |  11.0 KB  |  433 lines
  1. /*
  2.  
  3.   Nullsoft WASABI Source File License
  4.  
  5.   Copyright 1999-2001 Nullsoft, Inc.
  6.  
  7.     This software is provided 'as-is', without any express or implied
  8.     warranty.  In no event will the authors be held liable for any damages
  9.     arising from the use of this software.
  10.  
  11.     Permission is granted to anyone to use this software for any purpose,
  12.     including commercial applications, and to alter it and redistribute it
  13.     freely, subject to the following restrictions:
  14.  
  15.     1. The origin of this software must not be misrepresented; you must not
  16.        claim that you wrote the original software. If you use this software
  17.        in a product, an acknowledgment in the product documentation would be
  18.        appreciated but is not required.
  19.     2. Altered source versions must be plainly marked as such, and must not be
  20.        misrepresented as being the original software.
  21.     3. This notice may not be removed or altered from any source distribution.
  22.  
  23.  
  24.   Brennan Underwood
  25.   brennan@nullsoft.com
  26.  
  27. */
  28.  
  29. /*
  30. ** JNetLib
  31. ** Copyright (C) 2000-2001 Nullsoft, Inc.
  32. ** Author: Justin Frankel
  33. ** File: udpconnection.cpp - JNL UDP connection implementation
  34. ** License: see jnetlib.h
  35. */
  36.  
  37. #include "netinc.h"
  38. #include "util.h"
  39. #include "udpconnection.h"
  40.  
  41.  
  42. JNL_UDPConnection::JNL_UDPConnection(short incoming_port, JNL_AsyncDNS *dns, int sendbufsize, int recvbufsize)
  43. {
  44.   m_errorstr="";
  45.   if (dns == JNL_CONNECTION_AUTODNS)
  46.   {
  47.     m_dns=new JNL_AsyncDNS();
  48.     m_dns_owned=1;
  49.   }
  50.   else
  51.   {
  52.     m_dns=dns;
  53.     m_dns_owned=0;
  54.   }
  55.   m_recv_buffer_len=recvbufsize;
  56.   m_send_buffer_len=sendbufsize;
  57.   m_recv_buffer=(char*)malloc(m_recv_buffer_len);
  58.   m_send_buffer=(char*)malloc(m_send_buffer_len);
  59.   m_socket=-1;
  60.   memset(m_recv_buffer,0,recvbufsize);
  61.   memset(m_send_buffer,0,sendbufsize);
  62.   m_remote_port=0;
  63.   m_state=STATE_NOCONNECTION;
  64.   m_recv_len=m_recv_pos=0;
  65.   m_send_len=m_send_pos=0;
  66.   m_host[0]=0;
  67.   memset(&m_saddr,0,sizeof(struct sockaddr_in));
  68.   m_socket=::socket(PF_INET,SOCK_DGRAM,0);
  69.   if (m_socket==-1)
  70.   {
  71.     m_errorstr="creating socket";
  72.     m_state=STATE_ERROR;
  73.   }
  74.   SET_SOCK_BLOCK(m_socket,0);
  75.   memset(&m_iaddr,0,sizeof(struct sockaddr_in));
  76.   m_iaddr.sin_family=AF_INET;
  77.   m_iaddr.sin_port=htons(incoming_port);
  78.   m_iaddr.sin_addr.s_addr = htonl( INADDR_ANY );
  79.   if(::bind(m_socket,(struct sockaddr *)&m_iaddr,sizeof(m_iaddr))==-1)
  80.   {
  81.     m_errorstr="binding socket";
  82.     m_state=STATE_ERROR;
  83.   }
  84.   m_state=STATE_CONNECTED;
  85. }
  86.  
  87. void JNL_UDPConnection::setpeer(char *hostname, int port)
  88. {
  89.   m_remote_port=(short)port;
  90.   strncpy(m_host,hostname,sizeof(m_host)-1);
  91.   m_host[sizeof(m_host)-1]=0;
  92.   memset(&m_saddr,0,sizeof(m_saddr));
  93.   if (!m_host[0])
  94.   {
  95.     m_errorstr="empty hostname";
  96.     m_state=STATE_ERROR;
  97.   }
  98.   else
  99.   {
  100.     m_state=STATE_RESOLVING;
  101.     m_saddr.sin_family=AF_INET;
  102.     m_saddr.sin_port=htons((unsigned short)port);
  103.     m_saddr.sin_addr.s_addr=inet_addr(hostname);
  104.   }
  105. }
  106.  
  107. void JNL_UDPConnection::setpeer(struct sockaddr *addr)
  108. {
  109.   memcpy(&m_saddr, (const void *)addr, sizeof(struct sockaddr_in));
  110. }
  111.  
  112. JNL_UDPConnection::~JNL_UDPConnection()
  113. {
  114.   if (m_socket >= 0)
  115.   {
  116.     ::shutdown(m_socket, SHUT_RDWR);
  117.     ::closesocket(m_socket);
  118.     m_socket=-1;
  119.   }
  120.   free(m_recv_buffer);
  121.   free(m_send_buffer);
  122.   if (m_dns_owned) 
  123.   {
  124.     delete m_dns;
  125.   }
  126. }
  127.  
  128. void JNL_UDPConnection::run(int max_send_bytes, int max_recv_bytes, int *bytes_sent, int *bytes_rcvd)
  129. {
  130.   int bytes_allowed_to_send=(max_send_bytes<0)?m_send_buffer_len:max_send_bytes;
  131.   int bytes_allowed_to_recv=(max_recv_bytes<0)?m_recv_buffer_len:max_recv_bytes;
  132.  
  133.   if (bytes_sent) *bytes_sent=0;
  134.   if (bytes_rcvd) *bytes_rcvd=0;
  135.  
  136.   switch (m_state)
  137.   {
  138.     case STATE_RESOLVING:
  139.       if (m_saddr.sin_addr.s_addr == INADDR_NONE)
  140.       {
  141.         int a=m_dns?m_dns->resolve(m_host,(unsigned long int *)&m_saddr.sin_addr.s_addr):-1;
  142.         if (!a) { m_state=STATE_CONNECTED; }
  143.         else if (a == 1)
  144.         {
  145.           m_state=STATE_RESOLVING; 
  146.           break;
  147.         }
  148.         else
  149.         {
  150.           m_errorstr="resolving hostname"; 
  151.           m_state=STATE_ERROR; 
  152.           return;
  153.         }
  154.       }
  155.       m_state=STATE_CONNECTED;
  156.     break;
  157.     case STATE_CONNECTED:
  158.     case STATE_CLOSING:
  159.       if (m_send_len>0 && bytes_allowed_to_send>0)
  160.       {
  161.         int len=m_send_buffer_len-m_send_pos;
  162.         if (len > m_send_len) len=m_send_len;
  163.         if (len > bytes_allowed_to_send) len=bytes_allowed_to_send;
  164.         if (len > 0)
  165.         {
  166.           int res=::sendto(m_socket,m_send_buffer+m_send_pos,len,0,(struct sockaddr *)&m_saddr,sizeof(struct sockaddr_in));
  167.           if (res==-1 && ERRNO != EWOULDBLOCK)
  168.           {            
  169. //            m_state=STATE_CLOSED;
  170. //            return;
  171.           }
  172.           if (res>0)
  173.           {
  174.             bytes_allowed_to_send-=res;
  175.             if (bytes_sent) *bytes_sent+=res;
  176.             m_send_pos+=res;
  177.             m_send_len-=res;
  178.           }
  179.         }
  180.         if (m_send_pos>=m_send_buffer_len) 
  181.         {
  182.           m_send_pos=0;
  183.           if (m_send_len>0)
  184.           {
  185.             len=m_send_buffer_len-m_send_pos;
  186.             if (len > m_send_len) len=m_send_len;
  187.             if (len > bytes_allowed_to_send) len=bytes_allowed_to_send;
  188.             int res=::sendto(m_socket,m_send_buffer+m_send_pos,len,0,(struct sockaddr *)&m_saddr,sizeof(struct sockaddr_in));
  189.             if (res==-1 && ERRNO != EWOULDBLOCK)
  190.             {
  191. //              m_state=STATE_CLOSED;
  192.             }
  193.             if (res>0)
  194.             {
  195.               bytes_allowed_to_send-=res;
  196.               if (bytes_sent) *bytes_sent+=res;
  197.               m_send_pos+=res;
  198.               m_send_len-=res;
  199.             }
  200.           }
  201.         }
  202.       }
  203.       if (m_recv_len<m_recv_buffer_len)
  204.       {
  205.         int len=m_recv_buffer_len-m_recv_pos;
  206.         if (len > m_recv_buffer_len-m_recv_len) len=m_recv_buffer_len-m_recv_len;
  207.         if (len > bytes_allowed_to_recv) len=bytes_allowed_to_recv;
  208.         if (len>0)
  209.         {
  210.           int lg=sizeof(m_laddr);
  211.           int res=::recvfrom(m_socket,m_recv_buffer+m_recv_pos,len,0,(struct sockaddr *)&m_laddr,(socklen_t *)&lg);
  212.           if (res == 0 || (res < 0 && ERRNO != EWOULDBLOCK))
  213.           {        
  214.             m_state=STATE_CLOSED;
  215.             break;
  216.           }
  217.           if (res > 0)
  218.           {
  219.             bytes_allowed_to_recv-=res;
  220.             if (bytes_rcvd) *bytes_rcvd+=res;
  221.             m_recv_pos+=res;
  222.             m_recv_len+=res;
  223.           }
  224.         }
  225.         if (m_recv_pos >= m_recv_buffer_len)
  226.         {
  227.           m_recv_pos=0;
  228.           if (m_recv_len < m_recv_buffer_len)
  229.           {
  230.             len=m_recv_buffer_len-m_recv_len;
  231.             if (len > bytes_allowed_to_recv) len=bytes_allowed_to_recv;
  232.             if (len > 0)
  233.             {
  234.               int lg=sizeof(m_laddr);
  235.               int res=::recvfrom(m_socket,m_recv_buffer+m_recv_pos,len,0,(struct sockaddr *)&m_laddr,(socklen_t *)&lg);
  236.               if (res == 0 || (res < 0 && ERRNO != EWOULDBLOCK))
  237.               {        
  238.                 m_state=STATE_CLOSED;
  239.                 break;
  240.               }
  241.               if (res > 0)
  242.               {
  243.                 bytes_allowed_to_recv-=res;
  244.                 if (bytes_rcvd) *bytes_rcvd+=res;
  245.                 m_recv_pos+=res;
  246.                 m_recv_len+=res;
  247.               }
  248.             }
  249.           }
  250.         }
  251.       }
  252.       if (m_state == STATE_CLOSING)
  253.       {
  254.         if (m_send_len < 1) m_state = STATE_CLOSED;
  255.       }
  256.     break;
  257.     default: break;
  258.   }
  259. }
  260.  
  261. void JNL_UDPConnection::close(int quick)
  262. {
  263.   if (quick || m_state == STATE_RESOLVING)
  264.   {
  265.     m_state=STATE_CLOSED;
  266.     if (m_socket >= 0)
  267.     {
  268.       ::shutdown(m_socket, SHUT_RDWR);
  269.       ::closesocket(m_socket);
  270.     }
  271.     m_socket=-1;
  272.     memset(m_recv_buffer,0,m_recv_buffer_len);
  273.     memset(m_send_buffer,0,m_send_buffer_len);
  274.     m_remote_port=0;
  275.     m_recv_len=m_recv_pos=0;
  276.     m_send_len=m_send_pos=0;
  277.     m_host[0]=0;
  278.     memset(&m_saddr,0,NULL);
  279.   }
  280.   else
  281.   {
  282.     if (m_state == STATE_CONNECTED) m_state=STATE_CLOSING;
  283.   }
  284. }
  285.  
  286. int JNL_UDPConnection::send_bytes_in_queue(void)
  287. {
  288.   return m_send_len;
  289. }
  290.  
  291. int JNL_UDPConnection::send_bytes_available(void)
  292. {
  293.   return m_send_buffer_len-m_send_len;
  294. }
  295.  
  296. int JNL_UDPConnection::send(char *data, int length)
  297. {
  298.   if (length > send_bytes_available())
  299.   {
  300.     return -1;
  301.   }
  302.   
  303.   int write_pos=m_send_pos+m_send_len;
  304.   if (write_pos >= m_send_buffer_len) 
  305.   {
  306.     write_pos-=m_send_buffer_len;
  307.   }
  308.  
  309.   int len=m_send_buffer_len-write_pos;
  310.   if (len > length) 
  311.   {
  312.     len=length;
  313.   }
  314.  
  315.   memcpy(m_send_buffer+write_pos,data,len);
  316.   if (length > len)
  317.   {
  318.     memcpy(m_send_buffer,data+len,length-len);
  319.   }
  320.   m_send_len+=length;
  321.   return 0;
  322. }
  323.  
  324. int JNL_UDPConnection::send_string(char *line)
  325. {
  326.   return send(line,strlen(line));
  327. }
  328.  
  329. int JNL_UDPConnection::recv_bytes_available(void)
  330. {
  331.   return m_recv_len;
  332. }
  333.  
  334. int JNL_UDPConnection::peek_bytes(char *data, int maxlength)
  335. {
  336.   if (maxlength > m_recv_len)
  337.   {
  338.     maxlength=m_recv_len;
  339.   }
  340.   int read_pos=m_recv_pos-m_recv_len;
  341.   if (read_pos < 0) 
  342.   {
  343.     read_pos += m_recv_buffer_len;
  344.   }
  345.   int len=m_recv_buffer_len-read_pos;
  346.   if (len > maxlength)
  347.   {
  348.     len=maxlength;
  349.   }
  350.   memcpy(data,m_recv_buffer+read_pos,len);
  351.   if (len < maxlength)
  352.   {
  353.     memcpy(data+len,m_recv_buffer,maxlength-len);
  354.   }
  355.  
  356.   return maxlength;
  357. }
  358.  
  359. int JNL_UDPConnection::recv_bytes(char *data, int maxlength)
  360. {
  361.   
  362.   int ml=peek_bytes(data,maxlength);
  363.   m_recv_len-=ml;
  364.   return ml;
  365. }
  366.  
  367. int JNL_UDPConnection::getbfromrecv(int pos, int remove)
  368. {
  369.   int read_pos=m_recv_pos-m_recv_len + pos;
  370.   if (pos < 0 || pos > m_recv_len) return -1;
  371.   if (read_pos < 0) 
  372.   {
  373.     read_pos += m_recv_buffer_len;
  374.   }
  375.   if (read_pos >= m_recv_buffer_len)
  376.   {
  377.     read_pos-=m_recv_buffer_len;
  378.   }
  379.   if (remove) m_recv_len--;
  380.   return m_recv_buffer[read_pos];
  381. }
  382.  
  383. int JNL_UDPConnection::recv_lines_available(void)
  384. {
  385.   int l=recv_bytes_available();
  386.   int lcount=0;
  387.   int lastch=0;
  388.   int pos;
  389.   for (pos=0; pos < l; pos ++)
  390.   {
  391.     int t=getbfromrecv(pos,0);
  392.     if (t == -1) return lcount;
  393.     if ((t=='\r' || t=='\n') &&(
  394.          (lastch != '\r' && lastch != '\n') || lastch==t
  395.         )) lcount++;
  396.     lastch=t;
  397.   }
  398.   return lcount;
  399. }
  400.  
  401. int JNL_UDPConnection::recv_line(char *line, int maxlength)
  402. {
  403.   if (maxlength > m_recv_len) maxlength=m_recv_len;
  404.   while (maxlength--)
  405.   {
  406.     int t=getbfromrecv(0,1);
  407.     if (t == -1) 
  408.     {
  409.       *line=0;
  410.       return 0;
  411.     }
  412.     if (t == '\r' || t == '\n')
  413.     {
  414.       int r=getbfromrecv(0,0);
  415.       if ((r == '\r' || r == '\n') && r != t) getbfromrecv(0,1);
  416.       *line=0;
  417.       return 0;
  418.     }
  419.     *line++=(char)t;
  420.   }
  421.   return 1;
  422. }
  423.  
  424. unsigned long JNL_UDPConnection::get_interface(void)
  425. {
  426.   if (m_socket==-1) return 0;
  427.   struct sockaddr_in sin;
  428.   memset(&sin,0,sizeof(sin));
  429.   socklen_t len=16;
  430.   if (::getsockname(m_socket,(struct sockaddr *)&sin,&len)) return 0;
  431.   return (unsigned long) sin.sin_addr.s_addr;
  432. }
  433.